name: prompt-selector
endpoints:
- respond
- collect_goals
compose:
  env_file:
  - .env
  build:
    args:
      SERVICE_PORT: 8135
      SERVICE_NAME: prompt_selector
      SENTENCE_RANKER_SERVICE_URL: http://sentence-ranker:8128/respond
      N_SENTENCES_TO_RETURN: 2
      PROMPTS_TO_CONSIDER: dream_persona,casual_email,meeting_notes,official_email,plan_for_article
      FLASK_APP: server
    context: .
    dockerfile: ./annotators/prompt_selector/Dockerfile
  command: flask run -h 0.0.0.0 -p 8135
  environment:
  - FLASK_APP=server
  deploy:
    resources:
      limits:
        memory: 100M
      reservations:
        memory: 100M
  volumes:
  - ./annotators/prompt_selector:/src
  - ./common:/src/common
  ports:
  - 8135:8135
proxy: null
